/*
** $Id: print.c,v 1.54 2006/01/11 22:49:27 lhf Exp $
** print bytecodes
** See Copyright Notice in lua.h
*/

#include <ctype.h>
#include <stdio.h>

#define luac_c
#define LUA_CORE

#include "ldebug.h"
#include "lobject.h"
#include "lopcodes.h"
#include "lundump.h"

#define PrintFunction	luaU_print

#define Sizeof(x)	((int)sizeof(x))
#define VOID(p)		((const void*)(p))

static void PrintString(const Proto* f, int n)
{
	const char* s=svalue(&f->k[n]);
	putchar('"');
	for (; *s; s++)
	{
		switch (*s)
		{
		case '"': printf("\\\""); break;
		case '\a': printf("\\a"); break;
		case '\b': printf("\\b"); break;
		case '\f': printf("\\f"); break;
		case '\n': printf("\\n"); break;
		case '\r': printf("\\r"); break;
		case '\t': printf("\\t"); break;
		case '\v': printf("\\v"); break;
		default:	if (isprint((unsigned char)*s))
						printf("%c",*s);
					else
						printf("\\%03u",(unsigned char)*s);
		}
	}
	putchar('"');
}

static void PrintConstant(const Proto* f, int i)
{
	const TValue* o=&f->k[i];
	switch (ttype(o))
	{
	case LUA_TNIL:
		printf("nil");
		break;
	case LUA_TBOOLEAN:
		printf(bvalue(o) ? "true" : "false");
		break;
	case LUA_TNUMBER:
		printf(LUA_NUMBER_FMT,nvalue(o));
		break;
	case LUA_TSTRING:
		PrintString(f,i);
		break;
	default:				/* cannot happen */
		printf("? type=%d",ttype(o));
		break;
	}
}

static void PrintCode(const Proto* f)
{
	const Instruction* code=f->code;
	int pc,n=f->sizecode;
	for (pc=0; pc<n; pc++)
	{
		Instruction i=code[pc];
		OpCode o=GET_OPCODE(i);
		int a=GETARG_A(i);
		int b=GETARG_B(i);
		int c=GETARG_C(i);
		int bx=GETARG_Bx(i);
		int sbx=GETARG_sBx(i);
		int line=getline(f,pc);
		printf("\t%d\t",pc+1);
		if (line>0) printf("[%d]\t",line); else printf("[-]\t");
		printf("%-9s\t",luaP_opnames[o]);
		switch (getOpMode(o))
		{
		case iABC:
			printf("%d",a);
			if (getBMode(o)!=OpArgN) printf(" %d",ISK(b) ? (-1-INDEXK(b)) : b);
			if (getCMode(o)!=OpArgN) printf(" %d",ISK(c) ? (-1-INDEXK(c)) : c);
			break;
		case iABx:
			if (getBMode(o)==OpArgK) printf("%d %d",a,-1-bx); else printf("%d %d",a,bx);
			break;
		case iAsBx:
			if (o==OP_JMP) printf("%d",sbx); else printf("%d %d",a,sbx);
			break;
		}
		switch (o)
		{
		case OP_LOADK:
			printf("\t; "); PrintConstant(f,bx);
			break;
		case OP_GETUPVAL:
		case OP_SETUPVAL:
			printf("\t; %s", (f->sizeupvalues>0) ? getstr(f->upvalues[b]) : "-");
			break;
		case OP_GETGLOBAL:
		case OP_SETGLOBAL:
			printf("\t; %s",svalue(&f->k[bx]));
			break;
		case OP_GETTABLE:
		case OP_SELF:
			if (ISK(c)) { printf("\t; "); PrintConstant(f,INDEXK(c)); }
			break;
		case OP_SETTABLE:
		case OP_ADD:
		case OP_SUB:
		case OP_MUL:
		case OP_DIV:
		case OP_POW:
		case OP_EQ:
		case OP_LT:
		case OP_LE:
			if (ISK(b) || ISK(c))
			{
				printf("\t; ");
				if (ISK(b)) PrintConstant(f,INDEXK(b)); else printf("-");
				printf(" ");
				if (ISK(c)) PrintConstant(f,INDEXK(c)); else printf("-");
			}
			break;
		case OP_JMP:
		case OP_FORLOOP:
		case OP_FORPREP:
			printf("\t; to %d",sbx+pc+2);
			break;
		case OP_CLOSURE:
			printf("\t; %p",VOID(f->p[bx]));
			break;
		case OP_SETLIST:
			if (c==0) printf("\t; %d",(int)code[++pc]);
			else printf("\t; %d",c);
			break;
		default:
			break;
		}
		printf("\n");
	}
}

#define SS(x)	(x==1)?"":"s"
#define S(x)	x,SS(x)

static void PrintHeader(const Proto* f)
{
	const char* s=getstr(f->source);
	if (*s=='@' || *s=='=')
		s++;
	else if (*s==LUA_SIGNATURE[0])
		s="(bstring)";
	else
		s="(string)";
	printf("\n%s <%s:%d,%d> (%d instruction%s, %d bytes at %p)\n",
		(f->linedefined==0)?"main":"function",s,
		f->linedefined,f->lastlinedefined,
		S(f->sizecode),f->sizecode*Sizeof(Instruction),VOID(f));
	printf("%d%s param%s, %d slot%s, %d upvalue%s, ",
		f->numparams,f->is_vararg?"+":"",SS(f->numparams),
		S(f->maxstacksize),S(f->nups));
	printf("%d local%s, %d constant%s, %d function%s\n",
		S(f->sizelocvars),S(f->sizek),S(f->sizep));
}

static void PrintConstants(const Proto* f)
{
	int i,n=f->sizek;
	printf("constants (%d) for %p:\n",n,VOID(f));
	for (i=0; i<n; i++)
	{
		printf("\t%d\t",i+1);
		PrintConstant(f,i);
		printf("\n");
	}
}

static void PrintLocals(const Proto* f)
{
	int i,n=f->sizelocvars;
	printf("locals (%d) for %p:\n",n,VOID(f));
	for (i=0; i<n; i++)
	{
		printf("\t%d\t%s\t%d\t%d\n",
			i,getstr(f->locvars[i].varname),f->locvars[i].startpc+1,f->locvars[i].endpc+1);
	}
}

static void PrintUpvalues(const Proto* f)
{
	int i,n=f->sizeupvalues;
	printf("upvalues (%d) for %p:\n",n,VOID(f));
	if (f->upvalues==NULL) return;
	for (i=0; i<n; i++)
	{
		printf("\t%d\t%s\n",i,getstr(f->upvalues[i]));
	}
}

void PrintFunction(const Proto* f, int full)
{
	int i,n=f->sizep;
	PrintHeader(f);
	PrintCode(f);
	if (full)
	{
		PrintConstants(f);
		PrintLocals(f);
		PrintUpvalues(f);
	}
	for (i=0; i<n; i++) PrintFunction(f->p[i],full);
}
